"""
# -*- coding: utf-8 -*-
# @Time    : 2023/5/19 19:27
# @Author  : 王摇摆
# @FileName: ConfusionMatrix.py
# @Software: PyCharm
# @Blog    ：https://blog.csdn.net/weixin_44943389?type=blog
"""

# 绘画混淆矩阵

import matplotlib.pyplot as plt
import seaborn as sns

def plot_confusion_matrix(confusion_matrix, labels):
    # 设置图形大小
    plt.figure(figsize=(10, 8))

    # 使用seaborn库中的heatmap函数绘制热力图
    ax = sns.heatmap(confusion_matrix, annot=True, cmap="Blues", fmt="d")

    # 设置x轴和y轴的刻度标签
    ax.set_xticklabels(labels)
    ax.set_yticklabels(labels)

    # 设置x轴和y轴的标签名
    plt.xlabel('Predicted Labels')
    plt.ylabel('True Labels')

    # 设置图表标题
    plt.title('Confusion Matrix')

    # 显示图表
    plt.show()

# 示例混淆矩阵
labels = ['A', 'B', 'C']
confusion_matrix = [[13, 2, 3],
                    [1, 15, 2],
                    [4, 1, 12]]

# 绘制混淆矩阵热力图
plot_confusion_matrix(confusion_matrix, labels)
